package com.nuyoah.controller;


import cn.hutool.core.map.MapUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.nuyoah.common.dto.MenuDto;
import com.nuyoah.common.lang.Result;
import com.nuyoah.entity.Menu;
import com.nuyoah.entity.RoleMenu;
import com.nuyoah.entity.User;
import com.nuyoah.service.MenuService;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.util.StringUtils;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.security.Principal;
import java.time.LocalDateTime;
import java.util.List;

/**
 * 菜单相关接口
 *
 * @author xiaoming
 * @since 2024-03-26
 */
@RestController
@RequestMapping("/sys/menu")
public class MenuController extends BaseController{


    /**
     * 获取导航栏信息和用户权限信息
     * @param principal 当前登录用户
     * @return 返回包含导航栏信息和权限信息的结果对象
     */
    @GetMapping("/nav")
    public Result nav(Principal principal){
        User user = userService.getByUsername(principal.getName()); // 通过用户名获取用户信息

        // 获取权限信息
        String authorityInfo = userService.getUserAuthorityInfo(user.getId());
        String[] authorityInfoArray = StringUtils.tokenizeToStringArray(authorityInfo,",");

        // 获取导航栏信息
        List<MenuDto> navs = menuService.getCurrentUserNav();

        return Result.succ(MapUtil.builder()
                .put("authoritys",authorityInfoArray)
                .put("nav",navs)
                .map()
        );
    }

    /**
     * 根据菜单ID获取菜单信息
     * @param id 菜单ID
     * @return 返回获取到的菜单信息
     */
    @GetMapping("/info/{id}")
    @PreAuthorize("hasAuthority('sys:menu:list')")
    public Result info(@PathVariable(name = "id") Long id) {
        return Result.succ(menuService.getById(id));
    }

    /**
     * 获取菜单列表
     * @return 返回菜单列表
     */
    @GetMapping("/list")
    @PreAuthorize("hasAuthority('sys:menu:list')")
    public Result list() {
        List<Menu> menus = menuService.tree(); // 获取菜单树结构
        return Result.succ(menus);
    }

    /**
     * 保存菜单
     * @param sysMenu 要保存的菜单信息
     * @return 返回保存后的菜单信息
     */
    @PostMapping("/save")
    @PreAuthorize("hasAuthority('sys:menu:save')")
    public Result save(@Validated @RequestBody Menu sysMenu) {
        sysMenu.setCreated(LocalDateTime.now()); // 设置创建时间

        menuService.save(sysMenu);
        return Result.succ(sysMenu);
    }

    /**
     * 更新菜单
     * @param sysMenu 要更新的菜单信息
     * @return 返回更新后的菜单信息
     */
    @PostMapping("/update")
    @PreAuthorize("hasAuthority('sys:menu:update')")
    public Result update(@Validated @RequestBody Menu sysMenu) {
        sysMenu.setUpdated(LocalDateTime.now()); // 设置更新时间

        menuService.updateById(sysMenu);

        // 清除所有与该菜单相关的权限缓存
        userService.clearUserAuthorityInfoByMenuId(sysMenu.getId());
        return Result.succ(sysMenu);
    }

    /**
     * 根据菜单ID删除菜单
     * @param id 菜单ID
     * @return 返回删除结果
     */
    @PostMapping("/delete/{id}")
    @PreAuthorize("hasAuthority('sys:menu:delete')")
    public Result delete(@PathVariable("id") Long id) {
        int count = menuService.count(new QueryWrapper<Menu>().eq("parent_id", id)); // 检查是否有子菜单
        if (count > 0) {
            return Result.fail("请先删除子菜单");
        }

        // 清除所有与该菜单相关的权限缓存
        userService.clearUserAuthorityInfoByMenuId(id);

        menuService.removeById(id); // 删除菜单

        // 同步删除中间关联表
        roleMenuService.remove(new QueryWrapper<RoleMenu>().eq("menu_id", id));
        return Result.succ("");
    }

}
